/*
 * Licensed to the Tai Ping Jin Ke
 *
 * Copyright (c) 2022 .
 * All rights reserved.
 * 项目名称：太平金科-通用能力封装层-公共组件
 * 版权说明：本软件属太平金融科技服务(上海)有限公司所有，在未获得太平金融科技服务(上海)有限公司正式授权情况下，任何企业和个人，不能获取、阅读、安装、传播本软件涉及的任何受知识产权保护的内容。
 */
package com.taipingframework.utility.extend;

import com.taipingframework.utility.constant.ApplicationConstant;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;
import reactor.util.context.Context;

public class ReactorHelper {

    /**
     * 获取上下文对象
     */
    public static Mono<Context> getContext() {
        return Mono.deferWithContext(Mono::just);
    }

    /**
     * 获取服务网络交换器
     */
    public static Mono<ServerWebExchange> getExchange() {
        return getContext()
                .map(context -> context.get(ServerWebExchange.class));
    }

    /**
     * 获取当前Web服务的请求对象
     */
    public static Mono<ServerHttpRequest> getRequest() {
        return getExchange()
                .map(ServerWebExchange::getRequest);
    }

    /**
     * 获取当前Web服务的响应对象
     */
    public static Mono<ServerHttpResponse> getResponse() {
        return getExchange()
                .map(ServerWebExchange::getResponse);
    }

    /**
     * 获取当前Web服务的请求标识
     */
    public static Mono<String> getRequestId() {
        return getExchange()
                .map(exchange -> exchange.getAttribute(ApplicationConstant.RESPONSE_IDENTITY));
    }

}
